Python : assign a column using lambda with if else